package com.course.util;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.lang.reflect.Method;
import java.net.URLEncoder;
import java.text.DecimalFormat;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import javax.servlet.http.HttpServletResponse;
import org.apache.poi.hssf.usermodel.HSSFCellStyle;
import org.apache.poi.hssf.usermodel.HSSFFont;
import org.apache.poi.hssf.usermodel.HSSFPalette;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.CellStyle;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.util.CellRangeAddress;
public class PoiExcelExport {
    HttpServletResponse response;
    // 文件名
    private String fileName ;
    //文件保存路径
    private String fileDir;
    //sheet名
    private String sheetName;
    //表头字体
    private String titleFontType = "Arial Unicode MS";
    //表头背景色
    private String titleBackColor = "C1FBEE";
    //表头字号
    private short titleFontSize = 12;
    //添加自动筛选的列 如 A:M
    private String address = "";
    //正文字体
    private String contentFontType = "Arial Unicode MS";
    //正文字号
    private short contentFontSize = 12;
    //Float类型数据小数位
    private String floatDecimal = ".00";
    //Double类型数据小数位
    private String doubleDecimal = ".00";
    //设置列的公式
    private String colFormula[] = null;
    DecimalFormat floatDecimalFormat=new DecimalFormat(floatDecimal);
    DecimalFormat doubleDecimalFormat=new DecimalFormat(doubleDecimal);
    private HSSFWorkbook workbook = null;
    public PoiExcelExport(String fileDir,String sheetName){
        this.fileDir = fileDir;
        this.sheetName = sheetName;
        workbook = new HSSFWorkbook();
    }
    public PoiExcelExport(HttpServletResponse response,String fileName,String sheetName){
        this.response = response;
        this.sheetName = sheetName;
        this.fileName=fileName;
        workbook = new HSSFWorkbook();
    }
    /**
    * 设置表头字体.
    * @param titleFontType
    */
    public void setTitleFontType(String titleFontType) {
        this.titleFontType = titleFontType;
    }
    /**
    * 设置表头背景色.
    * @param titleBackColor 十六进制
    */
    public void setTitleBackColor(String titleBackColor) {
        this.titleBackColor = titleBackColor;
    }
    /**
    * 设置表头字体大小.
    * @param titleFontSize
    */
    public void setTitleFontSize(short titleFontSize) {
        this.titleFontSize = titleFontSize;
    }
    /**
    * 设置表头自动筛选栏位,如A:AC.
    * @param address
    */
    public void setAddress(String address) {
        this.address = address;
    }
    /**
    * 设置正文字体.
    * @param contentFontType
    */
    public void setContentFontType(String contentFontType) {
        this.contentFontType = contentFontType;
    }
    /**
    * 设置正文字号.
    * @param contentFontSize
    */
    public void setContentFontSize(short contentFontSize) {
        this.contentFontSize = contentFontSize;
    }
    /**
    * 设置float类型数据小数位 默认.00
    * @param doubleDecimal 如 ".00"
    */
    public void setDoubleDecimal(String doubleDecimal) {
        this.doubleDecimal = doubleDecimal;
    }
    /**
    * 设置doubel类型数据小数位 默认.00
    * @param floatDecimalFormat 如 ".00
    */
    public void setFloatDecimalFormat(DecimalFormat floatDecimalFormat) {
        this.floatDecimalFormat = floatDecimalFormat;
    }
    /**
    * 设置列的公式
    * @param colFormula  存储i-1列的公式 涉及到的行号使用@替换 如A@+B@
    */
    public void setColFormula(String[] colFormula) {
        this.colFormula = colFormula;
    }
    /**
    * 写excel.
    * @param titleColumn  对应bean的属性名
    * @param titleName   excel要导出的表名
    * @param titleSize   列宽
    * @param dataList  数据
    */
    public void wirteExcel(String titleColumn[],String titleName[],int titleSize[],List<Map<String,Object>> dataList){
        //添加Worksheet（不添加sheet时生成的xls文件打开时会报错)
        Sheet sheet = workbook.createSheet(this.sheetName);
        //新建文件
        OutputStream out = null;
        try {
            if(fileDir!=null){
                //有文件路径
                out = new FileOutputStream(fileDir);
                }else{
                    //否则，直接写到输出流中
                    out = response.getOutputStream();
                    fileName = fileName+".xls";
                    response.setContentType("application/x-msdownload");
                    response.setHeader("Content-Disposition", "attachment; filename="
                    + URLEncoder.encode(fileName, "UTF-8"));
                }
                //写入excel的表头
                Row titleNameRow = workbook.getSheet(sheetName).createRow(0);
                //设置样式
                HSSFCellStyle titleStyle = workbook.createCellStyle();
                titleStyle = (HSSFCellStyle) setFontAndBorder(titleStyle, titleFontType, (short) titleFontSize);
                titleStyle = (HSSFCellStyle) setColor(titleStyle, titleBackColor, (short)10);
                for(int i = 0;i < titleName.length;i++){
                    sheet.setColumnWidth(i, titleSize[i]*256);    //设置宽度
                    Cell cell = titleNameRow.createCell(i);
                    cell.setCellStyle(titleStyle);
                    cell.setCellValue(titleName[i].toString());
                }
                //为表头添加自动筛选
                if(!"".equals(address)){
                    CellRangeAddress c = (CellRangeAddress) CellRangeAddress.valueOf(address);
                    sheet.setAutoFilter(c);
                }
                //通过反射获取数据并写入到excel中
                if(dataList!=null&&dataList.size()>0){
                    //设置样式
                    HSSFCellStyle dataStyle = workbook.createCellStyle();
                    titleStyle = (HSSFCellStyle) setFontAndBorder(titleStyle, contentFontType, (short) contentFontSize);
                    if(titleColumn.length>0){
                        for(int rowIndex = 1;rowIndex<=dataList.size();rowIndex++){
                            Map<String,Object> obj = dataList.get(rowIndex-1);     //获得该对象
                            Row dataRow = workbook.getSheet(sheetName).createRow(rowIndex);
                            for(int columnIndex = 0;columnIndex<titleColumn.length;columnIndex++){
                                String title = titleColumn[columnIndex].toString().trim();
                                if(!"".equals(title)){  //字段不为空
                                    Object tmpData  =obj.get(titleColumn[columnIndex]);
                                    String data = tmpData==null?"":tmpData.toString();
                                    Cell cell = dataRow.createCell(columnIndex);
                                    if(data!=null&&!"".equals(data)){
                                        cell.setCellValue(data);
                                    }
                                    }else{   //字段为空 检查该列是否是公式
                                        if(colFormula!=null){
                                            String sixBuf = colFormula[columnIndex].replace("@", (rowIndex+1)+"");
                                            Cell cell = dataRow.createCell(columnIndex);
                                            cell.setCellFormula(sixBuf.toString());
                                        }
                                    }
                                }
                            }
                        }
                    }
                    workbook.write(out);
                    } catch (Exception e) {
                        e.printStackTrace();
                        } finally {
                            try {
                                out.close();
                                } catch (IOException e) {
                                    e.printStackTrace();
                                }
                            }
                        }
                        /**
                        * 将16进制的颜色代码写入样式中来设置颜色
                        * @param style  保证style统一
                        * @param color 颜色：66FFDD
                        * @param index 索引 8-64 使用时不可重复
                        * @return
                        */
                        public CellStyle setColor(CellStyle style,String color,short index){
                            if(color!=""&&color!=null){
                                //转为RGB码
                                int r = Integer.parseInt((color.substring(0,2)),16);   //转为16进制
                                int g = Integer.parseInt((color.substring(2,4)),16);
                                int b = Integer.parseInt((color.substring(4,6)),16);
                                //自定义cell颜色
                                HSSFPalette palette = workbook.getCustomPalette();
                                palette.setColorAtIndex((short)index, (byte) r, (byte) g, (byte) b);
                                style.setFillPattern(CellStyle.SOLID_FOREGROUND);
                                style.setFillForegroundColor(index);
                            }
                            return style;
                        }
                        /**
                        * 设置字体并加外边框
                        * @param style  样式
                        * @param style  字体名
                        * @param style  大小
                        * @return
                        */
                        public CellStyle setFontAndBorder(CellStyle style,String fontName,short size){
                            HSSFFont font = workbook.createFont();
                            font.setFontHeightInPoints(size);
                            font.setFontName(fontName);
                            font.setBold(true);
                            style.setFont(font);
                            style.setBorderBottom(CellStyle.BORDER_THIN); //下边框
                            style.setBorderLeft(CellStyle.BORDER_THIN);//左边框
                            style.setBorderTop(CellStyle.BORDER_THIN);//上边框
                            style.setBorderRight(CellStyle.BORDER_THIN);//右边框
                            return style;
                        }
                        /**
                        * 删除文件
                        * @param fileDir
                        * @return
                        */
                        public boolean deleteExcel(){
                            boolean flag = false;
                            File file = new File(this.fileDir);
                            // 判断目录或文件是否存在
                            if (!file.exists()) {  // 不存在返回 false
                                return flag;
                                } else {
                                    // 判断是否为文件
                                    if (file.isFile()) {  // 为文件时调用删除文件方法
                                        file.delete();
                                        flag = true;
                                    }
                                }
                                return flag;
                            }
                            /**
                            * 删除文件
                            * @param fileDir
                            * @return
                            */
                            public boolean deleteExcel(String path){
                                boolean flag = false;
                                File file = new File(path);
                                // 判断目录或文件是否存在
                                if (!file.exists()) {  // 不存在返回 false
                                    return flag;
                                    } else {
                                        // 判断是否为文件
                                        if (file.isFile()) {  // 为文件时调用删除文件方法
                                            file.delete();
                                            flag = true;
                                        }
                                    }
                                    return flag;
                                }
                                public static void main(String[] args) {
                                }
                            }
